# -*- coding:utf-8 -*-
from PIL import Image
import os

# 设置目标宽高值, 默认图像为横向
target_width = 640
target_height = 480

path = './'
files = os.listdir(path)

start_string = input('请输入图片前缀:')
count = 0

# 遍历文件集合中的所有文件
for file_name in files:

    # 仅针对其中的 PNG 或者 JPG 文件做处理
    if file_name.endswith(".jpg") or file_name.endswith(".png"):
    
        # 依次打开各个JPG文件
        original_image = Image.open(file_name)
        
        # 获取图像原始宽高
        original_width, original_height = original_image.size
        
        # 将图像缩放为 720P 大小
        if (original_width > original_height):
            resize_image = original_image.resize((target_width, target_height))
        else:
            resize_image = original_image.resize((target_height, target_width))

        # 缩放后记得到暂存, 不然不生效
        resize_image.save(file_name)

        # 计数变量, 用于图像重新命令的序号
        count = count + 1
        new_file_name = '%s%02d.png' % (start_string, count)
        print(file_name, '>>>', new_file_name)

        # 执行重命名动作
        os.rename(file_name, new_file_name)

print('done')